from django.db import models
from legendofkrouzt.personagem.models import Personagem
from string import capitalize


ATIVA = 'a'
INATIVA = 'i'
STATUS_CHOICE = (
        (ATIVA,'Ativa'),
        (INATIVA,'Inativa')
)

class Mensagem(models.Model):
    assunto = models.CharField(max_length=32)
    mensagem = models.TextField()
    remetente = models.ForeignKey(Personagem)
    destinatario = models.IntegerField()
    data = models.DateTimeField()
    status = models.CharField(
               max_length=1,
               default=ATIVA,
               choices=STATUS_CHOICE
    )
    
    def __unicode__(self):
        return capitalize(self.assunto)
